/*
   Copyright 2011 James Bowling

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/

package com.armedia.dctm.constraints;

import java.util.ArrayList;
import java.util.List;

import com.documentum.com.DfClientX;
import com.documentum.com.IDfClientX;
import com.documentum.fc.client.IDfCollection;
import com.documentum.fc.client.IDfPersistentObject;
import com.documentum.fc.client.IDfQuery;
import com.documentum.fc.client.IDfSession;
import com.documentum.fc.common.DfException;
import com.documentum.fc.common.DfId;
import com.documentum.fc.common.IDfId;

public class ForeignKeysForTypeQuery {
	
	public List<ForeignKeyDto> execute(IDfSession session, String typeName) throws DfException {
		final List<IDfId> foreignKeyIds = new ArrayList<IDfId>();
		final String queryPattern = "select r_object_id from dm_foreign_key where child_type = '%s'";
		String query = String.format(queryPattern, typeName);
		IDfClientX client = new DfClientX();
		IDfQuery q = client.getQuery();
		q.setDQL(query);
		IDfCollection coll = null;
		try {
			coll = q.execute(session, IDfQuery.READ_QUERY);
			while (coll.next()) {
				String objectId = coll.getString("r_object_id");
				IDfId id = new DfId(objectId);
				foreignKeyIds.add(id);
			}
			
		} finally {
			if (coll != null) {
				coll.close();
				coll = null;
			}
		}
		
		final List<ForeignKeyDto> foreignKeyDtos = new ArrayList<ForeignKeyDto>();		
		for (IDfId iDfId : foreignKeyIds) {
			final IDfPersistentObject object = session.getObject(iDfId);
			final ForeignKeyDto foreignKeyDto = transfer(object);
			foreignKeyDtos.add(foreignKeyDto);
		}
	
		return foreignKeyDtos;
	}

	private ForeignKeyDto transfer(IDfPersistentObject foreignKeyObject) throws DfException {
		final ForeignKeyDto foreignKeyDto = new ForeignKeyDto(); 
		final String childTypeName = foreignKeyObject.getString("child_type");
		foreignKeyDto.setChildTypeName(childTypeName);
		final String parentTypeName = foreignKeyObject.getString("parent_type");
		foreignKeyDto.setParentTypeName(parentTypeName);
		final int parentAttributesValueCount = foreignKeyObject.getValueCount("parent_attributes");
		
		for (int i = 0; i < parentAttributesValueCount; i++) {
			final String parentAttribute = foreignKeyObject.getRepeatingString("parent_attributes", i);
			foreignKeyDto.getParentAttributes().add(parentAttribute);
			final String childAttribute = foreignKeyObject.getRepeatingString("child_attributes", i);
			foreignKeyDto.getChildAttributes().add(childAttribute);
		}
		
		return foreignKeyDto;
	}

}
